package com.qianfeng.iot.service.impl;

//
//                            _ooOoo_  
//                           o8888888o  
//                           88" . "88  
//                           (| -_- |)  
//                            O\ = /O  
//                        ____/`---'\____  
//                      .   ' \\| |// `.  
//                       / \\||| : |||// \  
//                     / _||||| -:- |||||- \  
//                       | | \\\ - /// | |  
//                     | \_| ''\---/'' | |  
//                      \ .-\__ `-` ___/-. /  
//                   ___`. .' /--.--\ `. . __  
//                ."" '< `.___\_<|>_/___.' >'"".  
//               | | : `- \`.;`\ _ /`;.`/ - ` : | |  
//                 \ \ `-. \_ __\ /__ _/ .-` / /  
//         ======`-.____`-.___\_____/___.-`____.-'======  
//                            `=---='  
//  
//         .............................................  
//                  佛祖镇楼                  BUG辟易  
//          佛曰:  
//                  写字楼里写字间，写字间里程序员；  
//                  程序人员写程序，又拿程序换酒钱。  
//                  酒醒只在网上坐，酒醉还来网下眠；  
//                  酒醉酒醒日复日，网上网下年复年。  
//                  但愿老死电脑间，不愿鞠躬老板前；  
//                  奔驰宝马贵者趣，公交自行程序员。  
//                  别人笑我忒疯癫，我笑自己命太贱；  
//  


import com.dc3.common.bean.R;
import com.dc3.common.constant.CacheConstant;
import com.dc3.common.constant.ServiceConstant;
import com.dc3.common.utils.Dc3Util;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.qianfeng.iot.openfeign.CacheFeignClient;
import com.qianfeng.iot.service.TokenService;
import io.jsonwebtoken.Jwt;
import io.jsonwebtoken.Jwts;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.Map;

/**
 * Created by Jackiechan on 2022/8/2 20:06
 *
 * @author Jackiechan
 * @version 1.0
 * @since 1.0
 */
@Service
public class TokenServiceImpl implements TokenService {

    private ObjectMapper objectMapper;

    private CacheFeignClient cacheFeignClient;


    @Autowired
    public void setCacheFeignClient(CacheFeignClient cacheFeignClient) {
        this.cacheFeignClient = cacheFeignClient;
    }

    @Autowired
    public void setObjectMapper(ObjectMapper objectMapper) {
        this.objectMapper = objectMapper;
    }

    @Override
    public Mono<R<Object>> checkToken(ServerWebExchange exchange) {
        //我们需要获取到用户的令牌数据
        String jwtHeaderValue = exchange.getRequest().getHeaders().getFirst(ServiceConstant.Header.X_AUTH_TOKEN);
        //如果没有就不执行
        //通过解析令牌数据,得到里面的用户名+租户id
   return  Mono.just(jwtHeaderValue)
                .map(jwt -> jwt.split("\\."))//按照.进行分割
                .map(strings -> strings[1])//获取到下标为1的数据,也就是存放数据的那一部分
                .doOnError(throwable -> throwable.printStackTrace())//发生错误的时候会执行,我看看什么错误
                .map(body -> Dc3Util.decode(body))//将正文的base64转成原始的json
                .map(json -> {
                    try {
                        return objectMapper.readValue(json, Map.class);//转成map
                    } catch (IOException e) {
                        throw new RuntimeException(e);
                    }
                }).flatMap(map ->
                {
                    String sub = (String) map.get("sub");//获取到我们的用户名&租户id
                    String[] userNameTenantId = sub.split("&");//切成数组
                    //组成我们当前数据在redis中的key
                    String key = userNameTenantId[1] + CacheConstant.Suffix.TENANT_ID + CacheConstant.Entity.USER + userNameTenantId[0] + CacheConstant.Suffix.SALT;
                    return cacheFeignClient.getValue(key);//从redis中获取到盐
                }).map(slat -> {
                    try {
                        Jwt parse = Jwts.parser().setSigningKey(slat.getBytes(StandardCharsets.UTF_8)).parse(jwtHeaderValue);
                        //如果解析成功,说明我们的令牌没问题,
                        return R.ok();
                    } catch (Exception e) {
                        e.printStackTrace();
                        return R.fail("令牌校验失败");
                    }
                })
                //当出现错误或者数据为空的时候,我们都校验失败
                .onErrorReturn(R.fail("令牌格式不对")).defaultIfEmpty(R.fail("令牌校验失败"));
    }
}
